package com.banmoon.datastructure.SparseArray;

import java.util.Arrays;

/**
 * 稀疏数组
 *
 * @author banmoon
 */
public class SparseArray {

    public static void main(String[] args) {
        int[][] arrays = new int[11][11];
        arrays[2][3] = 1;
        arrays[2][4] = 2;
        arrays[3][3] = 1;
        for (int[] array : arrays) {
            System.out.println(Arrays.toString(array));
        }
        // 转换为稀疏数组
        System.out.println("================ 分割线 ================");
        int[][] sparseArray = generateSparseArray(arrays);
        for (int[] ints : sparseArray) {
            System.out.println(Arrays.toString(ints));
        }
        // 还原为原始数组
        System.out.println("================ 分割线 ================");
        int[][] originArray = restoreSparseArray(sparseArray);
        for (int[] ints : originArray) {
            System.out.println(Arrays.toString(ints));
        }
    }

    /**
     * 生成稀疏数组
     * @param originArray 原始数组
     * @return 稀疏数组
     */
    public static int[][] generateSparseArray(int[][] originArray) {
        // 得到原始数组的行个数、列个数
        int row = originArray.length;
        int col = originArray[0].length;
        // 得到原始数组的有效值个数
        int num = 0;
        for (int[] ints : originArray) {
            for (int j = 0; j < col; j++) {
                if (ints[j] != 0) {
                    num++;
                }
            }
        }
        // 创建稀疏数组，并设置第0行的数据
        int[][] sparseArray = new int[num + 1][3];
        sparseArray[0][0] = row;
        sparseArray[0][1] = col;
        sparseArray[0][2] = num;
        // 将剩余的有效值压缩至稀疏数组
        int currentRow = 1;
        for (int i = 0; i < row; i++) {
            for (int j = 0; j < col; j++) {
                if (originArray[i][j] != 0) {
                    sparseArray[currentRow][0] = i;
                    sparseArray[currentRow][1] = j;
                    sparseArray[currentRow][2] = originArray[i][j];
                    currentRow++;
                }
            }
        }
        return sparseArray;
    }

    /**
     * 还原稀疏数组
     * @param sparseArray 稀疏数组
     * @return 原始数组
     */
    public static int[][] restoreSparseArray(int[][] sparseArray) {
        int row = sparseArray[0][0];
        int col = sparseArray[0][1];
        // 创建原始数组
        int[][] originArray = new int[row][col];
        // 写入有效值
        for (int i = 1; i < sparseArray.length; i++) {
            int r = sparseArray[i][0];
            int c = sparseArray[i][1];
            int val = sparseArray[i][2];
            originArray[r][c] = val;
        }
        return originArray;
    }

}
